# -*- coding: utf-8 -*-
# 1.引入必须的包
import json
import os
import threading

import openai

# 2.获取api-key
openai.api_key = ''

mutex = threading.Lock()
def chat(text):
  # 3.使用OpenAI的API完成ChatGPT模型调用
  #    model：指的就是ChatGPT模型
  #    prompt：向ChatGPT提出的问题
  #    max_tokens：返回的最大字符个数
  with mutex:
    response = openai.Completion.create(
      model="text-davinci-003",
      prompt=text,
      max_tokens=2048,
    )
    text = ''
    for choice in response.choices:
      text = text + '\n' + choice.text
  return text

